from PIL import Image
import os

# 指定包含jpg文件的文件夹路径
source_folder = 'C:/Users/LZH/Desktop/code/code/data/vaihingen/img_dir/train'
# 指定保存png文件的文件夹路径
target_folder = 'C:/Users/LZH/Desktop/code/code/data/vaihingen/img_dir/train'

# 确保目标文件夹存在
os.makedirs(target_folder, exist_ok=True)

# 遍历源文件夹中的所有文件
for filename in os.listdir(source_folder):
    if filename.endswith('.jpg'):
        # 创建不带扩展名的文件名
        file_without_extension = os.path.splitext(filename)[0]
        # 读取图像
        img = Image.open(os.path.join(source_folder, filename))
        # 保存为png格式
        img.save(os.path.join(target_folder, file_without_extension + '.png'))

print("转换完成")